package com.study.direct.config;

import com.study.constant.Constant;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.*;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

@Configuration
@Slf4j
public class DirectRabbitConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnsCallback {

    @Resource
    private RabbitTemplate rabbitTemplate;

    @PostConstruct
    public void init() {
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnsCallback(this);
    }

    /**
     *
     * @描述：创建队列
     * @作者：严磊
     * @时间：2020年7月1日 下午2:49:41
     * @return
     */
    @Bean("infoLogQueue")
    public Queue infoLogQueue() {
        return new Queue(Constant.DirectConstant.INFO_LOG_QUEUE, true, false, false);
    }
    @Bean("errorLogQueue")
    public Queue errorLogQueue() {
        return new Queue(Constant.DirectConstant.ERROR_LOG_QUEUE, true, false, false);
    }

    /**
     *
     * @描述：创建Direct交换机
     * @作者：严磊
     * @时间：2020年7月1日 下午2:49:35
     * @return
     */
    @Bean("logDirectExchange")
    public DirectExchange logDirectExchange() {
        return new DirectExchange(Constant.DirectConstant.LOG_DIRECT_EXCHANGE, true, false);
    }

    /**
     *
     * @描述：将队列和交换机绑定, 并设置用于匹配的路由键
     * @作者：严磊
     * @时间：2020年7月1日 下午2:49:29
     * @param queue
     * @param exchange
     * @return
     */
    @Bean("infoLogBindingDirect")
    public Binding infoBindingDirect(@Qualifier(value = "infoLogQueue") Queue queue,
                                     @Qualifier(value = "logDirectExchange") DirectExchange exchange) {
        return BindingBuilder.bind(queue)
                .to(exchange)
                .with(Constant.DirectConstant.logLevel.info);
    }

    @Bean("errorLogBindingDirect")
    public Binding errorLogBindingDirect(@Qualifier(value = "errorLogQueue") Queue queue,
                                         @Qualifier(value = "logDirectExchange") DirectExchange exchange) {
        return BindingBuilder.bind(queue)
                .to(exchange)
                .with(Constant.DirectConstant.logLevel.error);
    }

    /**
     * 消息确认机制：确认消息提供者是否成功发送消息到交换机
     * 消息成功到达交换机触发该方法
     *
     * @param correlationData
     * @param ack
     * @param cause
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            //消息成功到达交换机
            log.info("消息成功到达交换机:消息id={}", correlationData.getId());
        } else {
            log.error("消息未到达交换机,消息id={}，原因={}", correlationData.getId(), cause);
        }
    }

    /**
     * return机制：确认消息是否成功的从交换机分发到队列
     * 配置publisher-returns为true 消息未成功到达队列，会触发该方法
     *
     * @param returnedMessage
     */
    @Override
    public void returnedMessage(ReturnedMessage returnedMessage) {
        log.error("消息未到达队列:", returnedMessage.toString());
    }
}
